table = {
    'jdata': {
        'edge_name': {2: 'buy', 1: 'view', 3: 'follow', 4: 'comment'},
        'target': [('user', 'buy', 'item')],
    },
    'yoochoose': {
        'edge_name': {1: 'buy', 2: 'click'},
        'target': [('user', 'buy', 'item')],
    },
    'yelp': {
        'target': [('user', 'review', 'item'), ('user', 'tip', 'item')],
        'knowledge': ['user-user'],
    },
    'Taobao': {
        # 'edge_name': {'buy': 'buy', 'pv': 'pv', 'cart': 'cart', 'fav': 'fav'},
        'target': [('user', 'buy', 'item')]
    },
    'ml-1m': {
        'edge_name': {1: 'rate', 2: 'rate', 3: 'rate', 4: 'rate', 5: 'rate',
                      6: 'belong'},
        'target': [('user', 'rate', 'item')],
        'features': ['user'],
        'knowledge': ['item-genre'],
        # 'co-user': 50,
        # 'co-item': 20
        # 'num_nei': {('user', 'rate', 'item'): -1,
        #             ('item', 'rate-ed', 'user'): -1,
        #             ('item', 'belong', 'genre'): 30,
        #             ('genre', 'belong-ed', 'item'): 30}
    },
    'ml-100k': {
        'edge_name': {1: 'rate', 2: 'rate', 3: 'rate', 4: 'rate', 5: 'rate',
                      6: 'belong'},
        'target': [('user', 'rate', 'item')],
        'features': ['user', 'item'],
        'knowledge': ['item-genre'],
        # 'co-user': 10,
        # 'co-item': 10
    },
    'twitter': {
        'target': [('user', 'MT', 'user'), ('user', 'RE', 'user'),
                   ('user', 'RT', 'user')],
    },
    'ia-movielens-user2tags-10m': {
        'edge_name': {1: 'tag'},
        'target': [('user', 'tag', 'item')],
        # 'co-user': 100,
        # 'co-item': 100
    }
}

hyperparams = {
    'emb_dim': 64,
    'hidden_dim': 64,
    'dropout': 0.8,
}

settings = {
    'reload': True,
    'n_epochs': 1,
    'lr': 0.001,
    'l2': 0,
    'ref': 0,
    'small_better': [False],
    'early_stop': 4,
    'show_progress': True,
}

statistics = {

}
